package edu.ptit.app;

import edu.ptit.jad.configuration.CONSTANTS;
import jgibblda.Estimator;
import jgibblda.Inferencer;
import jgibblda.LDACmdOption;
import jgibblda.Model;

public class AppLDAInferencer {

	public static void main(String[] args) {

		LDACmdOption ldaOption = new LDACmdOption();
		ldaOption.inf = false;
		ldaOption.est = true;
		ldaOption.dir = "models/onlyad";
		ldaOption.modelName = "model-final";
		ldaOption.niters = 2000;

		Inferencer inferencer = new Inferencer();
		ldaOption.dfile = "../../data/test/100_test_news.jad";
		inferencer.init(ldaOption);
		Model newModel = inferencer.inference();
		newModel.saveModel("news_only_ad");

		ldaOption.dfile = "../../data/test/2000_test_ads.jad";
		inferencer.init(ldaOption);
		newModel = inferencer.inference();
		newModel.saveModel("ads_only_ad");

	}

}
